#!/usr/bin/env python
# -*- conding: utf-8 -*-

"""
@Time     : 2024/8/20 21:14
@Author   : liujingmao
@File     : 3.消息提示模板拼接.py
"""

from langchain_core.prompts import ChatPromptTemplate

"""
1.从message中构建系统提示ChatPromptTemplate
"""
system_chat_prompt = ChatPromptTemplate.from_messages([
    ("system", "你是MAI开发的聊天机器人，请根据用户的提示进行回复，我叫{username}"),
])

"""
2.从message中构建用户类型的ChatPromptTemplate
"""
human_chat_prompt = ChatPromptTemplate.from_messages([
    ("human", "{query}")
])

"""
3. 将系统和用户类型的chat模板ChatPromptTemplate拼接
"""
chat_prompt = system_chat_prompt + human_chat_prompt

"""
4. system+user-->dict
"""

sys_user_dic = {"username":"liujingmao","query":"您好，我是茂茂公司开发的智能机器人，请问，有什么帮助您？"}

result = chat_prompt.invoke(sys_user_dic)

print("********----print result------***********")

print(result)


print("-----------------------------------")
print(chat_prompt.invoke({
    "username": "茂茂小智",
    "query": "你好，你是哪位？"
}))
